package com.jcfk.eam.service.base;

import com.jcfk.eam.domain.dto.mt.EamMtUseApplyDTO;
import com.jcfk.eam.domain.po.base.EamProductAssetModel;
import com.jcfk.eam.domain.po.main.EamAssetInfo;
import com.jcfk.eam.domain.query.base.EamProductAssetModelQuery;
import com.jcfk.eam.domain.dto.base.EamProductAssetModelDTO;
import com.jcfk.eam.dao.base.EamProductAssetModelDao;
import com.jcfk.common.business.service.BaseService;
import com.jcfk.common.web.domain.DataPage;
import com.jcfk.common.web.domain.RetResult;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.jcfk.eam.domain.uploadExcel.base.EamProductAssetModelUploadExcel;
import com.jcfk.eam.service.main.EamAssetInfoService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import org.springframework.transaction.annotation.Transactional;
import com.jcfk.common.core.utils.BeanUtils;
import cn.hutool.core.util.StrUtil;
import com.jcfk.common.security.utils.SecurityUtils;
import com.alibaba.druid.util.StringUtils;

import java.util.*;

import static com.jcfk.common.core.utils.BeanUtils.convertList;

import java.util.stream.Collectors;

/**
 * 产品与资产型号
 *
 * @author yql
 * @email 522891237@gqq.com
 * @date 2024-07-25
 */
@Service
public class EamProductAssetModelService extends BaseService<EamProductAssetModelDao, EamProductAssetModel> {

    @Autowired
    private EamAssetInfoService eamAssetInfoService;

    /**
     * EamProductAssetModel分页查询
     *
     * @param page  分页设置
     * @param query 查询条件
     * @return 分页查询结果
     */
    public DataPage<EamProductAssetModelDTO> queryByPage(DataPage<EamProductAssetModelDTO> page, EamProductAssetModelQuery query) {
        // 分页查询
        query.setProductName(StringUtils.isEmpty(query.getProductName()) ? query.getProductName() : query.getProductName() + "%");
        query.setAssetModelNo(StringUtils.isEmpty(query.getAssetModelNo()) ? query.getAssetModelNo() : query.getAssetModelNo() + "%");
        page = this.baseMapper.queryByPage(page, query);
        return page;
    }


    /**
     * EamProductAssetModel分页查询
     *
     * @param query 查询条件
     * @return 分页查询结果
     */
    public List<EamProductAssetModelDTO> selectAllList(EamProductAssetModelQuery query) {
        // 分页查询
        query.setProductName(StringUtils.isEmpty(query.getProductName()) ? query.getProductName() : query.getProductName() + "%");
        query.setAssetModelNo(StringUtils.isEmpty(query.getAssetModelNo()) ? query.getAssetModelNo() : query.getAssetModelNo() + "%");

        if (query.getLocationId() != null && !"".equals(query.getLocationId())) {
            QueryWrapper<EamAssetInfo> queryWrapper = new QueryWrapper<>();
            queryWrapper.lambda().eq(EamAssetInfo::getLocationId, query.getLocationId());
            List<EamAssetInfo> list = eamAssetInfoService.list(queryWrapper);
            List<String> ids = list.stream().map(EamAssetInfo::getAssetTypeId).collect(Collectors.toList());
            if (ids.size() <= 0) return new ArrayList<>();
            query.setIds(ids);
        }
        return this.baseMapper.selectAllList(query);
    }


    /**
     * EamProductAssetModel分页查询
     *
     * @param query 查询条件
     * @return 分页查询结果
     */
    public EamMtUseApplyDTO selectAllListByNowProductId(EamProductAssetModelQuery query) {
        EamMtUseApplyDTO eamMtUseApplyDTO = new EamMtUseApplyDTO();
        // 原来产品的
        EamProductAssetModelQuery originalEamProductAssetModelQuery = new EamProductAssetModelQuery();
        originalEamProductAssetModelQuery.setProductId(query.getOriginalProductId());
        originalEamProductAssetModelQuery.setLocationId(query.getLocationId());
        List<EamProductAssetModelDTO> originalEamProductAssetModelDTOS = new ArrayList<>();
        if (query.getLocationId() != null && !"".equals(query.getLocationId())) {
            QueryWrapper<EamAssetInfo> queryWrapper = new QueryWrapper<>();
            queryWrapper.lambda().eq(EamAssetInfo::getLocationId, query.getLocationId());
            List<EamAssetInfo> list = eamAssetInfoService.list(queryWrapper);
            List<String> ids = list.stream().map(EamAssetInfo::getAssetTypeId).collect(Collectors.toList());
            if (ids.size() > 0) {
                originalEamProductAssetModelQuery.setIds(ids);
                originalEamProductAssetModelDTOS = this.baseMapper.selectAllList(originalEamProductAssetModelQuery);
            }
        } else {
            originalEamProductAssetModelDTOS = this.baseMapper.selectAllList(originalEamProductAssetModelQuery);
        }
        eamMtUseApplyDTO.setCurrentDetails(originalEamProductAssetModelDTOS);


        List<EamProductAssetModelDTO> eamProductAssetModelDTOS = new ArrayList<>();
        // 当前产品的
        EamProductAssetModelQuery eamProductAssetModelQuery = new EamProductAssetModelQuery();
        eamProductAssetModelQuery.setProductId(query.getProductId());
        eamProductAssetModelQuery.setLocationId(query.getLocationId());
        if (query.getLocationId() != null && !"".equals(query.getLocationId())) {
            QueryWrapper<EamAssetInfo> queryWrapper = new QueryWrapper<>();
            queryWrapper.lambda().eq(EamAssetInfo::getLocationId, query.getLocationId());
            List<EamAssetInfo> list = eamAssetInfoService.list(queryWrapper);
            List<String> ids = list.stream().map(EamAssetInfo::getAssetTypeId).collect(Collectors.toList());
            if (ids.size() > 0) {
                eamProductAssetModelQuery.setIds(ids);
                eamProductAssetModelDTOS = this.baseMapper.selectAllList(eamProductAssetModelQuery);
            }
        } else {
            eamProductAssetModelDTOS = this.baseMapper.selectAllList(eamProductAssetModelQuery);
        }

        eamMtUseApplyDTO.setChangeDetails(eamProductAssetModelDTOS);

        if (eamProductAssetModelDTOS.size() == 0 || originalEamProductAssetModelDTOS.size() == 0)
            return eamMtUseApplyDTO;


        List<EamProductAssetModelDTO> retoriginalEamProductAssetModelDTOS = new ArrayList<>();
        List<String> types = new ArrayList<>();
        for (EamProductAssetModelDTO item : eamProductAssetModelDTOS) {
            if (item.getAssetTypeId() != null) {
                for (EamProductAssetModelDTO listItem : originalEamProductAssetModelDTOS) {
                    if (item.getAssetTypeId().equals(listItem.getAssetTypeId())) {
                        retoriginalEamProductAssetModelDTOS.add(item);
                        types.add(item.getAssetTypeId());
                        break;
                    }
                }
            }
        }

        for (EamProductAssetModelDTO item : originalEamProductAssetModelDTOS) {
            if (!types.contains(item.getAssetTypeId())) {
                retoriginalEamProductAssetModelDTOS.add(item);
            }
        }

        Map<String, Integer> assetTypeIndexMap = new HashMap<>();
        for (int i = 0; i < retoriginalEamProductAssetModelDTOS.size(); i++) {
            assetTypeIndexMap.put(retoriginalEamProductAssetModelDTOS.get(i).getAssetTypeId(), i);
        }

        eamMtUseApplyDTO.setCurrentDetails(retoriginalEamProductAssetModelDTOS);
        List<EamProductAssetModelDTO> others = new ArrayList<>();

        for (int i = 0; i < eamProductAssetModelDTOS.size(); ) {
            String assetTypeId = eamProductAssetModelDTOS.get(i).getAssetTypeId();
            if (assetTypeIndexMap.containsKey(assetTypeId)) {
                int index = assetTypeIndexMap.get(assetTypeId);
                if (index < eamProductAssetModelDTOS.size()) {
                    EamProductAssetModelDTO eamProductAssetModelDTO = eamProductAssetModelDTOS.remove(i);
                    eamProductAssetModelDTOS.add(index, eamProductAssetModelDTO);
                }
            } else {
                i++;
            }
        }
        eamMtUseApplyDTO.setChangeDetails(eamProductAssetModelDTOS);

        for (EamProductAssetModelDTO eamProductAssetModelDTO : eamProductAssetModelDTOS) {
            if (!assetTypeIndexMap.containsKey(eamProductAssetModelDTO.getAssetTypeId())) {
                others.add(eamProductAssetModelDTO);
            }
        }

        eamMtUseApplyDTO.setApplyDetails(others);

        return eamMtUseApplyDTO;
    }


    /**
     * 判断EamProductAssetModel是已经存在
     *
     * @param dto
     * @return
     */
    public RetResult<String> isValid(EamProductAssetModelDTO dto) {
        QueryWrapper<EamProductAssetModel> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("deleted", 0);

        return RetResult.ok();
    }

    /**
     * 插入产品与资产型号
     *
     * @param dto 插入对象
     * @return
     */
    @Transactional(rollbackFor = Exception.class)
    public void insert(EamProductAssetModelDTO dto) {

        // 保存
        EamProductAssetModel item = new EamProductAssetModel();
        BeanUtils.copyProperties(dto, item);
        this.setDefaultValue(item);

        super.save(item);
    }

    /**
     * 更新产品与资产型号
     *
     * @param dto 更新对象
     * @return
     */
    @Transactional(rollbackFor = Exception.class)
    public void update(EamProductAssetModelDTO dto) {

        // 保存
        EamProductAssetModel item = new EamProductAssetModel();
        BeanUtils.copyProperties(dto, item);


        // 默认数据
        this.setDefaultValue(item);

        super.updateById(item);
    }

    /**
     * 根据批量主键删除产品与资产型号(逻辑删除)
     *
     * @param dto
     * @return bool
     */
    @Transactional(rollbackFor = Exception.class)
    public Boolean logicDeleteList(EamProductAssetModelDTO dto) {

        EamProductAssetModel item = new EamProductAssetModel();

        this.setDefaultValue(item);
        item.setVersion(dto.getVersion());
        item.setDeleted(1);

        QueryWrapper<EamProductAssetModel> queryWrapper = new QueryWrapper<>();
        queryWrapper.in("id", dto.getIds());

        return this.update(item, queryWrapper);
    }

    /**
     * 设置添加默认值
     *
     * @param item 添加对象
     */
    public void setDefaultValue(EamProductAssetModel item) {
        if (StrUtil.isEmpty(item.getCreatedUserId())) {
            item.setCreatedUserId(SecurityUtils.getUserId());
            item.setCreateTime(new Date());
            item.setVersion(1L);
            item.setDeleted(0);
            item.setCreatedUserName(SecurityUtils.getUserName());
        }
        item.setLastUpdatedUserId(SecurityUtils.getUserId());
        item.setLastUpdateTime(new Date());

        item.setLastUpdatedUserName(SecurityUtils.getUserName());

    }

    /**
     * 根据主键获取产品与资产型号信息
     *
     * @param id 主键
     * @return StandardCodeDTO
     */
    public EamProductAssetModelDTO get(String id) {
        // 查询数据
        EamProductAssetModel item = this.baseMapper.selectById(id);
        EamProductAssetModelDTO dto = new EamProductAssetModelDTO();
        try {
            BeanUtils.copyProperties(item, dto);
        } catch (RuntimeException e) {
            throw e;
        }

        return dto;
    }


    /**
     * 插入产品与资产型号
     *
     * @param list 批量插入产品与资产型号
     * @return
     */
    @Transactional(rollbackFor = Exception.class)
    public void insertBatch(List<EamProductAssetModelUploadExcel> list) {

        List<EamProductAssetModel> savelist = new ArrayList<>();
        list.forEach(itemOne -> {
            // 保存
            EamProductAssetModel item = new EamProductAssetModel();
            BeanUtils.copyProperties(itemOne, item);
            this.setDefaultValue(item);
            savelist.add(item);
        });
        // 保存
        if (savelist.size() > 0) super.saveBigBatch(savelist);
    }

    /**
     * 根据条件获取单表信息列表
     *
     * @param query 查询条件
     * @return List<EamProductAssetModelDTO>
     */
    public List<EamProductAssetModelDTO> getList(EamProductAssetModelQuery query) {
        // 查询条件
        QueryWrapper<EamProductAssetModel> qw = new QueryWrapper<>();
        qw.lambda().eq(!StrUtil.isEmpty(query.getPlantId()), EamProductAssetModel::getPlantId, query.getPlantId());
        qw.lambda().eq(!StrUtil.isEmpty(query.getProductId()), EamProductAssetModel::getProductId, query.getProductId());
        qw.lambda().likeRight(!StrUtil.isEmpty(query.getProductName()), EamProductAssetModel::getProductName, query.getProductName());
        qw.lambda().eq(query.getAssetKind() != null, EamProductAssetModel::getAssetKind, query.getAssetKind());
        qw.lambda().eq(!StrUtil.isEmpty(query.getAssetTypeId()), EamProductAssetModel::getAssetTypeId, query.getAssetTypeId());
        qw.lambda().eq(!StrUtil.isEmpty(query.getAssetModelId()), EamProductAssetModel::getAssetModelId, query.getAssetModelId());
        qw.lambda().likeRight(!StrUtil.isEmpty(query.getAssetModelNo()), EamProductAssetModel::getAssetModelNo, query.getAssetModelNo());
        qw.lambda().eq(query.getEnabled() != null, EamProductAssetModel::getEnabled, query.getEnabled());
        qw.lambda().eq(query.getDeleted() != null, EamProductAssetModel::getDeleted, query.getDeleted());

        // 查询数据
        List<EamProductAssetModel> list = super.list(qw);

        return convertList(list, EamProductAssetModelDTO.class);
    }

    /**
     * 根据主键获取产品与资产型号信息
     *
     * @param id 主键
     * @return StandardCodeDTO
     */
    public void deleteitem(String id) {
        EamProductAssetModel item = getById(id);
        this.setDefaultValue(item);
        item.setDeleted(1);

        this.updateById(item);
    }
}

